import type { NodeType, Schema } from "@rivet/model";
// 获取节点类型
export function getNodeType(
  nameOrType: string | NodeType,
  schema: Schema,
): NodeType {
  if (typeof nameOrType === "string") {
    if (!schema.nodes[nameOrType]) {
      throw Error(`没有找到节点类型是'${nameOrType}'. 肯能是你忘记添加扩展了?`);
    }

    return schema.nodes[nameOrType];
  }

  return nameOrType;
}
